import os
import django

# 设置Django设置模块
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'ip_locator.settings')
# 初始化Django
django.setup()

# 导入用户模型
from django.contrib.auth.models import User

try:
    # 尝试获取admin用户
    u = User.objects.get(username='admin')
    print(f'用户admin存在，ID: {u.id}')
    # 重置密码
    u.set_password('admin123')
    u.save()
    print('密码已重置为: admin123')
except User.DoesNotExist:
    # 如果用户不存在，创建新的超级用户
    print('用户admin不存在')
    User.objects.create_superuser('admin', 'admin@example.com', 'admin123')
    print('已创建admin用户，密码: admin123')
